import java.util.HashMap;
import java.util.Map;

class Solution {


    /**
     * 子数组的和可以被k整除的子数组的个数 974题
     * @param nums
     * @param k
     * @return
     */

    public int subarraysDivByK(int[] nums, int k) {
        Map<Integer,Integer> map = new HashMap<>();
        map.put(0,1);

        int sum = 0, count = 0;
        for(int x : nums){
            sum += x;// 计算前缀和
            int num = (sum%k + k) % k; // 修正一下这个数 负数模正数为负数
            count += map.getOrDefault(num,0); // 计算一下num的次数
            map.put(num, map.getOrDefault(num,0)+1); // 更新结果
        }
        return count;
    }

}